/*
 * Copyright © 2018 www.noark.xyz All Rights Reserved.
 *
 * 感谢您选择Noark框架，希望我们的努力能为您提供一个简单、易用、稳定的服务器端框架 ！
 * 除非符合Noark许可协议，否则不得使用该文件，您可以下载许可协议文件：
 *
 *        http://www.noark.xyz/LICENSE
 *
 * 1.未经许可，任何公司及个人不得以任何方式或理由对本框架进行修改、使用和传播;
 * 2.禁止在本项目或任何子项目的基础上发展任何派生版本、修改版本或第三方版本;
 * 3.无论你对源代码做出任何修改和改进，版权都归Noark研发团队所有，我们保留所有权利;
 * 4.凡侵犯Noark版权等知识产权的，必依法追究其法律责任，特此郑重法律声明！
 */
package xyz.noark.network.rpc.stub;

import xyz.noark.core.network.PacketCodec;
import xyz.noark.core.thread.AsyncHelper;
import xyz.noark.core.thread.task.TaskQueue;
import xyz.noark.log.MDC;
import xyz.noark.network.codec.rpc.RpcPacket;
import xyz.noark.network.rpc.RpcCallback;

import java.io.Serializable;

/**
 * RPC异步请求的存根.
 *
 * @author 小流氓[176543888@qq.com]
 * @since 4.0
 */
public class RpcAsyncStub<T> extends RpcStub {
    private final Class<T> ackClass;
    private final RpcCallback<T> callback;
    private final Serializable queueId;
    private final long startTime;

    public RpcAsyncStub(PacketCodec packetCodec, Class<T> ackClass, RpcCallback<T> callback) {
        super(packetCodec);
        this.ackClass = ackClass;
        this.callback = callback;
        this.queueId = (Serializable) MDC.get(TaskQueue.QUEUE_ID);
        this.startTime = System.nanoTime();
    }

    @Override
    public void onResponse(RpcPacket packet) {
        // 消息回来了
        T protocol = packetCodec.decodeProtocol(packet.getByteArray(), ackClass);
        // 这里要切线程
        AsyncHelper.call(queueId, () -> callback.onSuccess(protocol));
    }
}